package JZOffer;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

/**
 * @author Lw中
 * @date 2020/8/6 11:35
 */
public class Jz32 {

    // Jz32.从上到下打印二叉树

     public class TreeNode {
          int val;
          TreeNode left;
          TreeNode right;
          TreeNode(int x) { val = x; }
      }

    /**
     * 使用广度优先搜索的方法完成树的遍历
      * @param root
     * @return List<List<Integer>>
     */
//    public List<List<Integer>> levelOrder(TreeNode root) {
//        // 二维数组
//        List<List<Integer>> lists = new ArrayList<List<Integer>>();
//        // 新建一个用链实现的队列用于存储节点
//        Queue<TreeNode> queue = new LinkedList<TreeNode>();
//        // 判断是否为空二叉树
//        if(root != null) {
//            queue.add(root);
//        }
//        while (!queue.isEmpty()) {
//            //创建arrayList用于存储每一层节点
//            List<Integer> list = new ArrayList<>();
//            for (int i = queue.size(); i > 0 ; i--) {
//                TreeNode node = queue.poll();
//                list.add(node.val);
//                // 如果左节点不为空入列
//                if (node.left != null) {
//                    queue.offer(node.left);
//                }
//                // 如果右节点不为空入列
//                if (node.right != null) {
//                    queue.offer(node.right);
//                }
//            }
//            lists.add(list);
//        }
//        return lists;
//    }

    /**
     * 使用广度优先搜索的方法完成树的遍历
     * @param root
     * @return int[]
     */
    public int[] levelOrder(TreeNode root) {
        if(root == null) return new int[0];
        // 创建一个队列用于存放节点以及遍历存放节点的子节点，先将根节点放入
        Queue<TreeNode> queue = new LinkedList<TreeNode>(){{ add(root); }};
        ArrayList<Integer> ans = new ArrayList<>();
        while(!queue.isEmpty()) {
            TreeNode node = queue.poll();
            ans.add(node.val);
            if(node.left != null) queue.add(node.left);
            if(node.right != null) queue.add(node.right);
        }
        int[] res = new int[ans.size()];
        for(int i = 0; i < ans.size(); i++)
            res[i] = ans.get(i);
        return res;
    }

}
